home *** CD-ROM | disk | FTP | other *** search
/ Enter 2006 September / Enter 09 2006.iso / Internet / SpamExperts Home 1.1 / SpamExperts Home.exe / lib / spamexperts.modules / email / Generator.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2006-07-14  |  9.8 KB  |  299 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. '''Classes to generate plain text from a message object tree.'''
  5. import re
  6. import sys
  7. import time
  8. import random
  9. import warnings
  10. from cStringIO import StringIO
  11. from email.Header import Header
  12. UNDERSCORE = '_'
  13. NL = '\n'
  14. fcre = re.compile('^From ', re.MULTILINE)
  15.  
  16. def _is8bitstring(s):
  17.     if isinstance(s, str):
  18.         
  19.         try:
  20.             unicode(s, 'us-ascii')
  21.         except UnicodeError:
  22.             return True
  23.         except:
  24.             None<EXCEPTION MATCH>UnicodeError
  25.         
  26.  
  27.     None<EXCEPTION MATCH>UnicodeError
  28.     return False
  29.  
  30.  
  31. class Generator:
  32.     '''Generates output from a Message object tree.
  33.  
  34.     This basic generator writes the message to the given file object as plain
  35.     text.
  36.     '''
  37.     
  38.     def __init__(self, outfp, mangle_from_ = True, maxheaderlen = 78):
  39.         """Create the generator for message flattening.
  40.  
  41.         outfp is the output file-like object for writing the message to.  It
  42.         must have a write() method.
  43.  
  44.         Optional mangle_from_ is a flag that, when True (the default), escapes
  45.         From_ lines in the body of the message by putting a `>' in front of
  46.         them.
  47.  
  48.         Optional maxheaderlen specifies the longest length for a non-continued
  49.         header.  When a header line is longer (in characters, with tabs
  50.         expanded to 8 spaces) than maxheaderlen, the header will split as
  51.         defined in the Header class.  Set maxheaderlen to zero to disable
  52.         header wrapping.  The default is 78, as recommended (but not required)
  53.         by RFC 2822.
  54.         """
  55.         self._fp = outfp
  56.         self._mangle_from_ = mangle_from_
  57.         self._maxheaderlen = maxheaderlen
  58.  
  59.     
  60.     def write(self, s):
  61.         self._fp.write(s)
  62.  
  63.     
  64.     def flatten(self, msg, unixfrom = False):
  65.         """Print the message object tree rooted at msg to the output file
  66.         specified when the Generator instance was created.
  67.  
  68.         unixfrom is a flag that forces the printing of a Unix From_ delimiter
  69.         before the first object in the message tree.  If the original message
  70.         has no From_ delimiter, a `standard' one is crafted.  By default, this
  71.         is False to inhibit the printing of any From_ delimiter.
  72.  
  73.         Note that for subobjects, no From_ line is printed.
  74.         """
  75.         if unixfrom:
  76.             ufrom = msg.get_unixfrom()
  77.             if not ufrom:
  78.                 ufrom = 'From nobody ' + time.ctime(time.time())
  79.             
  80.             print >>self._fp, ufrom
  81.         
  82.         self._write(msg)
  83.  
  84.     
  85.     def __call__(self, msg, unixfrom = False):
  86.         warnings.warn('__call__() deprecated; use flatten()', DeprecationWarning, 2)
  87.         self.flatten(msg, unixfrom)
  88.  
  89.     
  90.     def clone(self, fp):
  91.         '''Clone this generator with the exact same options.'''
  92.         return self.__class__(fp, self._mangle_from_, self._maxheaderlen)
  93.  
  94.     
  95.     def _write(self, msg):
  96.         oldfp = self._fp
  97.         
  98.         try:
  99.             self._fp = sfp = StringIO()
  100.             self._dispatch(msg)
  101.         finally:
  102.             self._fp = oldfp
  103.  
  104.         meth = getattr(msg, '_write_headers', None)
  105.         if meth is None:
  106.             self._write_headers(msg)
  107.         else:
  108.             meth(self)
  109.         self._fp.write(sfp.getvalue())
  110.  
  111.     
  112.     def _dispatch(self, msg):
  113.         main = msg.get_content_maintype()
  114.         sub = msg.get_content_subtype()
  115.         specific = UNDERSCORE.join((main, sub)).replace('-', '_')
  116.         meth = getattr(self, '_handle_' + specific, None)
  117.         if meth is None:
  118.             generic = main.replace('-', '_')
  119.             meth = getattr(self, '_handle_' + generic, None)
  120.             if meth is None:
  121.                 meth = self._writeBody
  122.             
  123.         
  124.         meth(msg)
  125.  
  126.     
  127.     def _write_headers(self, msg):
  128.         for h, v in msg.items():
  129.             print >>self._fp, '%s:' % h,
  130.             if self._maxheaderlen == 0:
  131.                 print >>self._fp, v
  132.                 continue
  133.             if isinstance(v, Header):
  134.                 print >>self._fp, v.encode()
  135.                 continue
  136.             if _is8bitstring(v):
  137.                 print >>self._fp, v
  138.                 continue
  139.             print >>self._fp, Header(v, maxlinelen = self._maxheaderlen, header_name = h, continuation_ws = '\t').encode()
  140.         
  141.         print >>self._fp
  142.  
  143.     
  144.     def _handle_text(self, msg):
  145.         payload = msg.get_payload()
  146.         if payload is None:
  147.             return None
  148.         
  149.         if not isinstance(payload, basestring):
  150.             raise TypeError('string payload expected: %s' % type(payload))
  151.         
  152.         if self._mangle_from_:
  153.             payload = fcre.sub('>From ', payload)
  154.         
  155.         self._fp.write(payload)
  156.  
  157.     _writeBody = _handle_text
  158.     
  159.     def _handle_multipart(self, msg):
  160.         msgtexts = []
  161.         subparts = msg.get_payload()
  162.         if subparts is None:
  163.             subparts = []
  164.         elif isinstance(subparts, basestring):
  165.             self._fp.write(subparts)
  166.             return None
  167.         elif not isinstance(subparts, list):
  168.             subparts = [
  169.                 subparts]
  170.         
  171.         for part in subparts:
  172.             s = StringIO()
  173.             g = self.clone(s)
  174.             g.flatten(part, unixfrom = False)
  175.             msgtexts.append(s.getvalue())
  176.         
  177.         alltext = NL.join(msgtexts)
  178.         boundary = msg.get_boundary(failobj = _make_boundary(alltext))
  179.         if msg.get_boundary() != boundary:
  180.             msg.set_boundary(boundary)
  181.         
  182.         if msg.preamble is not None:
  183.             print >>self._fp, msg.preamble
  184.         
  185.         print >>self._fp, '--' + boundary
  186.         if msgtexts:
  187.             self._fp.write(msgtexts.pop(0))
  188.         
  189.         for body_part in msgtexts:
  190.             print >>self._fp, '\n--' + boundary
  191.             self._fp.write(body_part)
  192.         
  193.         self._fp.write('\n--' + boundary + '--')
  194.         if msg.epilogue is not None:
  195.             print >>self._fp
  196.             self._fp.write(msg.epilogue)
  197.         
  198.  
  199.     
  200.     def _handle_message_delivery_status(self, msg):
  201.         blocks = []
  202.         for part in msg.get_payload():
  203.             s = StringIO()
  204.             g = self.clone(s)
  205.             g.flatten(part, unixfrom = False)
  206.             text = s.getvalue()
  207.             lines = text.split('\n')
  208.             if lines and lines[-1] == '':
  209.                 blocks.append(NL.join(lines[:-1]))
  210.                 continue
  211.             blocks.append(text)
  212.         
  213.         self._fp.write(NL.join(blocks))
  214.  
  215.     
  216.     def _handle_message(self, msg):
  217.         s = StringIO()
  218.         g = self.clone(s)
  219.         g.flatten(msg.get_payload(0), unixfrom = False)
  220.         self._fp.write(s.getvalue())
  221.  
  222.  
  223. _FMT = '[Non-text (%(type)s) part of message omitted, filename %(filename)s]'
  224.  
  225. class DecodedGenerator(Generator):
  226.     '''Generator a text representation of a message.
  227.  
  228.     Like the Generator base class, except that non-text parts are substituted
  229.     with a format string representing the part.
  230.     '''
  231.     
  232.     def __init__(self, outfp, mangle_from_ = True, maxheaderlen = 78, fmt = None):
  233.         """Like Generator.__init__() except that an additional optional
  234.         argument is allowed.
  235.  
  236.         Walks through all subparts of a message.  If the subpart is of main
  237.         type `text', then it prints the decoded payload of the subpart.
  238.  
  239.         Otherwise, fmt is a format string that is used instead of the message
  240.         payload.  fmt is expanded with the following keywords (in
  241.         %(keyword)s format):
  242.  
  243.         type       : Full MIME type of the non-text part
  244.         maintype   : Main MIME type of the non-text part
  245.         subtype    : Sub-MIME type of the non-text part
  246.         filename   : Filename of the non-text part
  247.         description: Description associated with the non-text part
  248.         encoding   : Content transfer encoding of the non-text part
  249.  
  250.         The default value for fmt is None, meaning
  251.  
  252.         [Non-text (%(type)s) part of message omitted, filename %(filename)s]
  253.         """
  254.         Generator.__init__(self, outfp, mangle_from_, maxheaderlen)
  255.         if fmt is None:
  256.             self._fmt = _FMT
  257.         else:
  258.             self._fmt = fmt
  259.  
  260.     
  261.     def _dispatch(self, msg):
  262.         for part in msg.walk():
  263.             maintype = part.get_content_maintype()
  264.             if maintype == 'text':
  265.                 print >>self, part.get_payload(decode = True)
  266.                 continue
  267.             if maintype == 'multipart':
  268.                 continue
  269.             print >>self, self._fmt % {
  270.                 'type': part.get_content_type(),
  271.                 'maintype': part.get_content_maintype(),
  272.                 'subtype': part.get_content_subtype(),
  273.                 'filename': part.get_filename('[no filename]'),
  274.                 'description': part.get('Content-Description', '[no description]'),
  275.                 'encoding': part.get('Content-Transfer-Encoding', '[no encoding]') }
  276.         
  277.  
  278.  
  279. _width = len(repr(sys.maxint - 1))
  280. _fmt = '%%0%dd' % _width
  281.  
  282. def _make_boundary(text = None):
  283.     token = random.randrange(sys.maxint)
  284.     boundary = '=' * 15 + _fmt % token + '=='
  285.     if text is None:
  286.         return boundary
  287.     
  288.     b = boundary
  289.     counter = 0
  290.     while True:
  291.         cre = re.compile('^--' + re.escape(b) + '(--)?$', re.MULTILINE)
  292.         if not cre.search(text):
  293.             break
  294.         
  295.         b = boundary + '.' + str(counter)
  296.         counter += 1
  297.     return b
  298.  
  299.